import tensorflow as tf
import tf2onnx

model = tf.keras.models.load_model("Model.keras")

for layer in model.layers:
    if isinstance(layer, tf.keras.layers.LSTM) or isinstance(layer, tf.keras.layers.GRU):
        layer.implementation = 1  # 强制使用标准实现（非 CuDNN）
        layer.unroll = True  # 展开循环（适用于短序列）

onnx_model, _ = tf2onnx.convert.from_keras(
    model,
    opset=13,
    output_path="Model.onnx"
)
